from data import *
import os.path as osp
import torch.utils.data as data
from utils.augmentations import SSDTestAugmentation
dataset = VOCDetection(root=osp.join(HOME, "data/pollen_data_multi_classes/test"), transform=SSDTestAugmentation(300, MEANS))
data_loader = data.DataLoader(dataset, 16,
                              num_workers=4,
                              shuffle=True, collate_fn=detection_collate,
                              pin_memory=True)

for i in data_loader:
    print(i)
    break
